package com.greedy.utils;

import java.util.Random;

/**
 * @Author: Yanxy
 * @Date: Created in:2022-12-24
 * Description: 验证码生成
 */
public class ValidateCodeUtil {

    /**　
     * 生成验证码
     * @param length 4/6位
     * @return 4/6位验证码
     */
    public static Integer generateValidateCode(int length) {
        Integer code = null;
        if (length == 4) {
            // 生成随机数最大9999
            code = new Random().nextInt(9999);
            // 确保随机数为4位数
            if (code < 1000) {
                code += 1000;
            }
        } else if (length == 6) {
            // 生成随机数最大999999
            code = new Random().nextInt(999999);
            if (code < 100000) {
                code += 100000;
            }
        }
        return code;
    }

    /**
     * 指定长度生成验证码
     * @param length 指定长度
     * @return 指定长度验证码
     */
    public static String generateValidateCodeByLength(int length) {
        return Integer.toHexString(new Random().nextInt()).substring(0, length);
    }
}
